import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.Arguments;
import org.junit.jupiter.params.provider.MethodSource;

import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.stream.Stream;

/**
 * 存在重复元素II
 * 题目：给定一个整数数组和一个整数k，判断数组中是否存在两个不同的索引i和j，使得nums [i] = nums [j]，并且 i 和 j的差的 绝对值 至多为 k。
 * <p>
 * 示例1:
 * 输入: nums = [1,2,3,1], k = 3
 * 输出: true
 * <p>
 * 示例 2:
 * 输入: nums = [1,0,1,1], k = 1
 * 输出: true
 * <p>
 * 示例 3:
 * 输入: nums = [1,2,3,1,2,3], k = 2
 * 输出: false
 * <p>
 * 来源：力扣（LeetCode-219）
 * 链接：https://leetcode-cn.com/problems/contains-duplicate-ii
 *
 * @author godfrey
 * @since 2020-12-03
 */
@DisplayName("存在重复元素II")
public class ContainsNearbyDuplicate extends BaseTest {

    static Stream<Arguments> testArguments() {
        return Stream.of(
                Arguments.arguments(new int[]{1, 2, 3, 1}, 3),
                Arguments.arguments(new int[]{1, 0, 1, 1}, 1),
                Arguments.arguments(new int[]{1, 2, 3, 1, 2, 3}, 2)
        );
    }

    @DisplayName("哈希表-时间复杂度O(n)，空间复杂度O(n)")
    @ParameterizedTest
    @MethodSource("testArguments")
    void containsNearbyDuplicate(int[] nums, int k) {
        final Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; ++i) {
            if (map.containsKey(nums[i])) {
                if (Math.abs(map.get(nums[i]) - i) <= k) {
                    System.out.println(true);
                    return;
                } else {
                    map.remove(nums[i]);
                    map.put(nums[i], i);
                }
            } else {
                map.put(nums[i], i);
            }
        }
        System.out.println(false);
    }

    @DisplayName("哈希表同时限制长度为k-时间复杂度O(n)，空间复杂度O(n)")
    @ParameterizedTest
    @MethodSource("testArguments")
    void containsNearbyDuplicate2(int[] nums, int k) {
        final HashSet<Integer> set = new HashSet<>();
        for (int i = 0; i < nums.length; i++) {
            if (set.contains(nums[i])) {
                System.out.println(true);
                return;
            }
            set.add(nums[i]);
            if (set.size() > k) {
                set.remove(nums[i - k]);
            }
        }
        System.out.println(false);
    }
}
